//给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。 
//
// k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。 
//
// 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。 
//
// 
//
// 示例 1： 
// 
// 
//输入：head = [1,2,3,4,5], k = 2
//输出：[2,1,4,3,5]
// 
//
// 示例 2： 
//
// 
//
// 
//输入：head = [1,2,3,4,5], k = 3
//输出：[3,2,1,4,5]
// 
//
// 
//提示：
//
// 
// 链表中的节点数目为 n 
// 1 <= k <= n <= 5000 
// 0 <= ListNode.val <= 1000
//
//
//
//
// 进阶：你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗？
//
//
//
//
// Related Topics 递归 链表 👍 2254 👎 0


package com.tyrone.leetcode.editor.cn;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Stack;

public class ReverseNodesInKGroup{

    public static void main(String[] args) {
        Solution solution = new ReverseNodesInKGroup().new Solution();
        int[] arr = new int[]{1,2,3,4,5};
        ListNode listNode = changeArrayToNode(arr);
        solution.reverseKGroup(listNode,2);
    }
    public static ListNode changeArrayToNode(int[] arr) {
        if(arr == null || arr.length <= 0){return null;}

        ListNode sentineNode = new ListNode(0);  //建立哨兵节点
        ListNode tempNode = sentineNode;  //tempNode变量指向哨兵节点
        for(int i = 0; i < arr.length; i++) {
            ListNode newNode = new ListNode(arr[i]);
            tempNode.next = newNode;    //临时节点‘后继指针’指向新节点
            tempNode = newNode;        //把新节点赋给临时变量tempNode
        }
        return sentineNode.next;
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    /**
     * Definition for singly-linked list.
     * public class ListNode {
     *     int val;
     *     ListNode next;
     *     ListNode() {}
     *     ListNode(int val) { this.val = val; }
     *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
     * }
     */

    /**
     *
     * 总体思路就是借助stack来处理每k个数据，接下来就是24题的套路了（并且要提前计算出长度）
     */
    class Solution {
        Stack<ListNode> stack  = new Stack<>();
        public ListNode reverseKGroup(ListNode head, int k) {
            if (head == null) return null;
            ListNode temp = head;
            int total = 0;
            while (temp!=null){
                total++;
                temp = temp.next;
            }

            return swapPairs(head,total,k);
        }

        private ListNode swapPairs(ListNode head, int total, int k) {
            if (total<k){
                return head;
            }

            ListNode tmp = head;
            for (int i = 0; i < k; i++) {
                stack.push(tmp);
                tmp = tmp.next;
            }
            //此时p1指向最后一个元素
            ListNode first = stack.pop();
            total--;
            ListNode p1 = first; //
            while (!stack.isEmpty()){
                ListNode pop = stack.pop();
                first.next = pop;
                first = first.next;
                total--;
            }
            if (total == 0) first.next = null;
            //当前k个已经处理完并且first指向当前k的最后一个
            first.next = swapPairs(tmp,total,k);
            return p1;
        }
    }
    //leetcode submit region end(Prohibit modification and deletion)
    static public class ListNode {
        int val;
        ListNode next;
        ListNode() {}
        ListNode(int val) { this.val = val; }
        ListNode(int val, ListNode next) { this.val = val; this.next = next; }
    }
}